import unittest
from time import sleep
from selenium import webdriver
from Page import Page

class Test_Case(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Firefox()
        self.driver.maximize_window()

    def tearDown(self):
        sleep(10)
        self.driver.quit()

    def test_baidu_01(self):
        po = Page(self.driver)
        po.baidu()
if __name__ == '__main__':
    unittest.main()
